7f5ca0651646f7cee286609e8c90f6984686a356,test-framework/performance/src/main/java/com/continuuity/performance/runner/PerformanceTestRunner.java,PerformanceTestRunner,init,#,280

Before Change


    }

    Module dataFabricModule;
    if (config.get("perf.datafabric.mode") != null
      && config.get("perf.datafabric.mode").equals("distributed")) {
      dataFabricModule = new DataFabricModules().getDistributedModules();
    } else {
      dataFabricModule = new DataFabricModules().getSingleNodeModules();
    }

    try {
      injector = Guice
        .createInjector(dataFabricModule,
                        new ConfigModule(config),
                        new IOModule(),
                        new LocationRuntimeModule().getInMemoryModules(),
                        new DiscoveryRuntimeModule().getInMemoryModules(),
                        new ProgramRunnerRuntimeModule().getInMemoryModules(),
                        new MetricsClientRuntimeModule().getInMemoryModules(),

After Change


    Module metricsClientModule;

    try {
      if (config.get("perf.reactor.mode") != null
        && config.get("perf.reactor.mode").equals("distributed")) {
        dataFabricModule = new DataFabricModules().getDistributedModules();
        locationModule = new LocationRuntimeModule().getDistributedModules();
        programRunnerModule = new ProgramRunnerRuntimeModule().getDistributedModules();
        metricsClientModule = new MetricsClientRuntimeModule().getInMemoryModules();
        zkClientService =
          ZKClientServices.delegate(
            ZKClients.reWatchOnExpire(
              ZKClients.retryOnFailure(
                ZKClientService.Builder.of(
                  config.get(Constants.CFG_ZOOKEEPER_ENSEMBLE)).setSessionTimeout(10000).build(),
                RetryStrategies.fixDelay(2, TimeUnit.SECONDS))));
        discoveryServiceModule = new DiscoveryRuntimeModule(zkClientService).getDistributedModules();
      } else {
        dataFabricModule = new DataFabricModules().getSingleNodeModules();
        locationModule = new LocationRuntimeModule().getInMemoryModules();
        programRunnerModule = new ProgramRunnerRuntimeModule().getInMemoryModules();
        metricsClientModule = new MetricsClientRuntimeModule().getInMemoryModules();
        discoveryServiceModule = new AbstractModule() {